import numpy as np
import matplotlib.pyplot as plt

# 参数方程绘制爱心
t = np.linspace(0, 2 * np.pi, 1000)

x = 16 * np.sin(t) ** 3
y = (13 * np.cos(t) -
     5 * np.cos(2 * t) -
     2 * np.cos(3 * t) -
     np.cos(4 * t))

plt.figure(figsize=(6,6))
plt.plot(x, y, color='red')
plt.fill(x, y, color='pink', alpha=0.6)
plt.title('❤️ PT ❤️ WLM ❤️', fontsize=16, color='red')
plt.axis('equal')
plt.axis('off')

# 保存成 jpg 文件
plt.savefig('heart.jpg', format='jpg', bbox_inches='tight', dpi=300)

# plt.show()
